/*
 * Copyright 2023 OPPO.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.oppo.cloud.parser.config;


import com.oppo.cloud.common.domain.cluster.hadoop.NameNodeConf;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * hadoop hdfs NameNode config
 */
@Configuration
@ConfigurationProperties(prefix = "hadoop")
@Data
@EnableConfigurationProperties(HadoopConfig.class)
public class HadoopConfig {

    private List<NameNodeConf> namenodes;

    public static final String NAME_NODE_MAP = "nameNodeMap";

    @Bean(name = NAME_NODE_MAP)
    public Map<String, NameNodeConf> loadNameNodeMap() {
        Map<String, NameNodeConf> nameNodeMap = new HashMap<>();
        for (NameNodeConf nameNode : namenodes) {
            nameNodeMap.put(nameNode.getNameservices(), nameNode);
        }
        return nameNodeMap;
    }
}
